/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package meminin.naivebayes;

import java.util.HashMap;

/**
 *
 * @author Danang
 */
public class NaiveBayesModel {
    HashMap<String, NaiveBayesClassModel> probability;
    
    public NaiveBayesModel() {
        probability = new HashMap<String, NaiveBayesClassModel>();
    }
    
    public void addAttribValueProbability(String attribValue, NaiveBayesClassModel nbc) {
        probability.put(attribValue, nbc);
    }
    
    public NaiveBayesClassModel getProbabilityFromAttribValue(String attribValue) {
        return probability.get(attribValue);
    }
    
}
